package com.hc.programming.tree;

import java.util.Arrays;
import java.util.LinkedList;
import java.util.Queue;

/**
 * 给你一棵二叉树的根节点 root ，翻转这棵二叉树，并返回其根节点。
 * <p>
 * 示例 1：
 * <a href="./翻转二叉树-示例1.png">示例1</a>
 * 输入：root = [4,2,7,1,3,6,9]
 * 输出：[4,7,2,9,6,3,1]
 * 示例 2：
 * 输入：root = [2,1,3]
 * 输出：[2,3,1]
 * 示例 3：
 * 输入：root = []
 * 输出：[]
 * <p>
 * 提示：
 * 树中节点数目范围在 [0, 100] 内
 * -100 <= Node.val <= 100
 *
 * @author huangchao E-mail:fengquan8866@163.com
 * @version 创建时间：2024/9/11 20:28
 */
public class 翻转二叉树 {
    public static void main(String[] args) {
        Integer[] arr = new Integer[]{4, 2, 7, 1, 3, 6, 9};
        TreeNode tree = TreeNode.tree(arr);
        System.out.println(Arrays.toString(arr) + "=[4,2,7,1,3,6,9],--" + invertTree(tree));
    }

    public static TreeNode invertTree(TreeNode root) {
//        return 递归(root);
        return 迭代(root);
    }

    private static TreeNode 迭代(TreeNode root) {
        if (root == null) return null;
        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(root);
        while (!queue.isEmpty()) {
            TreeNode node = queue.poll();
            if (node.left != null || node.right != null) {
                TreeNode temp = node.left;
                node.left = node.right;
                node.right = temp;
            }
            if (node.left != null) queue.offer(node.left);
            if (node.right != null) queue.offer(node.right);
        }
        return root;
    }

    private static TreeNode 递归(TreeNode root) {
        if (root == null) return root;
        TreeNode temp = root.left;
        root.left = root.right;
        root.right = temp;
        递归(root.left);
        递归(root.right);
        return root;
    }
}
